package com.wheat.common.core.controller;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.wheat.common.constant.Constants;
import com.wheat.common.page.PageBean;
import com.wheat.common.page.TableDataInfo;
import com.wheat.common.utils.security.ServletUtils;
import com.wheat.common.utils.sql.SqlUtil;

import java.util.List;

/**
 * 设置请求分页
 *
 * @author wheat on 2019/12/12.
 */
public class PageController {

    /**
     * 设置请求分页数据
     */
    public static void startPage() {
        PageBean pageDomain = buildPageRequest();
        Integer pageNum = pageDomain.getPageNum();
        Integer pageSize = pageDomain.getPageSize();
        if (pageNum != null && pageSize != null) {
            String orderBy = SqlUtil.escapeOrderBySql(pageDomain.getOrderBy());
            PageHelper.startPage(pageNum, pageSize, orderBy);
        }
    }

    /**
     * 封装分页对象
     */
    public static PageBean getPageDomain() {
        PageBean pageDomain = new PageBean();
        pageDomain.setPageNum(ServletUtils.getParameterToInt(Constants.PAGE_NUM));
        pageDomain.setPageSize(ServletUtils.getParameterToInt(Constants.PAGE_SIZE));
        pageDomain.setOrderByColumn(ServletUtils.getParameter(Constants.ORDER_BY_COLUMN));
        pageDomain.setIsAsc(ServletUtils.getParameter(Constants.IS_ASC));
        return pageDomain;
    }

    public static PageBean buildPageRequest() {
        return getPageDomain();
    }

    /**
     * 响应请求分页数据
     */
    @SuppressWarnings({"rawtypes", "unchecked"})
    public static TableDataInfo getDataTable(List<?> list) {
        TableDataInfo rspData = new TableDataInfo();
        rspData.setCode(0);
        rspData.setRows(list);
        rspData.setTotal(new PageInfo(list).getTotal());
        return rspData;
    }
}
